package com.xiaoyilin.interceptor;

import com.xiaoyilin.domain.Company;
import com.xiaoyilin.serializer.CompanyDeserializer;
import org.apache.kafka.clients.consumer.ConsumerConfig;
import org.apache.kafka.clients.consumer.ConsumerRecord;
import org.apache.kafka.clients.consumer.ConsumerRecords;
import org.apache.kafka.clients.consumer.KafkaConsumer;

import java.time.Duration;
import java.util.Collections;
import java.util.Properties;

/**
 * KafkaConsumer 非线程安全
 * <p>
 * KafkaConsumer 非线程安全并不意味着我们在消费消息的时候只能以单线程的方式执行。
 * 如果生产者发送消息的速度大于消费者处理消息的速度，
 * 由于 Kafka 中消息保留机制的作用，有些消息有可能在被消费之前就被清理了，从而造成消息的丢失 。
 * 我们可以通过多线程的方式来实现消息消费，
 * 第一种也是最常见的方式 ： 线程封闭，即为每个线程实例化一个 KafkaConsumer 对象
 * 第二种方式是多个消费线程同时消费同一个分区 ，这个通过 assign（）、 seek（）等 方法实现，这样可以打破原有的消费线程的个数不能超过分区数的限制，进一步提高了消费的 能力
 */
public class Consumer {

    public static final String topic = "test";
    public static final String brokerList = "localhost:9092";
    public static final String groupId = "myGroup";

    public static void main(String[] args) {
        Properties properties = new Properties();
        properties.put(ConsumerConfig.KEY_DESERIALIZER_CLASS_CONFIG, "org.apache.kafka.common.serialization.StringDeserializer");
        properties.put(ConsumerConfig.VALUE_DESERIALIZER_CLASS_CONFIG, CompanyDeserializer.class.getName());
        //消费者拦截器
        properties.put(ConsumerConfig.INTERCEPTOR_CLASSES_CONFIG, MyConsumerlnterceptor.class.getName());
        properties.put("bootstrap.servers", brokerList);
        properties.put("group.id", groupId);
        KafkaConsumer<String, Company> kafkaConsumer = new KafkaConsumer<>(properties);
        kafkaConsumer.subscribe(Collections.singletonList(topic));
        while (true) {
            ConsumerRecords<String, Company> consumerRecords = kafkaConsumer.poll(Duration.ofMillis(3000));
            for (ConsumerRecord<String, Company> record : consumerRecords) {
                System.out.println(record.value());
            }
        }
    }


}
